# -*-coding:utf-8-*-
"""
@project: python-project
@Time: 2024-07-25 11:08
@Author: Loneliness_Burial
@File: numpy增加或删除一个维度.py
@Motto: 深情不改必坠死海
@software: pycharm
@version: python 3.7
"""

import numpy as np

# 增加维度
arr = np.random.randint(1, 10, (3,))
print("创建一个一维数组：\n", arr)
print(arr.shape)
arr2 = np.expand_dims(arr, 0)
print("将一维数组变为二维数组： \n", arr2)
print(arr2.shape)
arr3 = np.expand_dims(np.expand_dims(arr, 0), 0)
print("将一维数组变为三维数组: \n", arr3)
print(arr3.shape)

# 删除维度
arr4 = np.squeeze(arr3, 0)
print("将三维数组变为二维数组: \n", arr4)
print(arr4.shape)
arr5 = np.squeeze(np.squeeze(arr3, 0), 0)
print("将三维数组变为一维数组： \n", arr5)
print(arr5.shape)

#
arr6 = arr[np.newaxis, :]
print("在索引为0的位置增加一个维度，长度为1； \n", arr6)
arr7 = arr[:, np.newaxis]
print("在索引为1的位置增加一个维度，长度为1： \n", arr7)
